Python interview coding question